import os
import qianfan

class TestQianfanSDK():
    def setup(self):
        os.environ["QIANFAN_ACCESS_KEY"] = "ALTAKiYLpYbzfKuouawYXLiBVs"
        os.environ["QIANFAN_SECRET_KEY"] = "1b19545e4646467eb690acae9806bbc5"
    def teardown_method(self):
        pass

    def testQianfanSDKchat(self):
        # 接下来就可以调用 SDK 的所有功能
        chat = qianfan.ChatCompletion()
        resp = chat.do(messages=[{"role": "user", "content": "人有几条腿"}])

        print(resp["result"])
        assert "有两条腿" in resp["result"]